# -*- coding:utf8 -*-
import json
import random

from squirrel_core.commons.utils.logger import Logging
from squirrel_core.commons.utils.tools import key_business_param_from_settings

__all__ = ["UserAgentMiddleware"]


class UserAgentMiddleware(object):
    """This middleware allows spiders to override the user_agent"""
    user_agent_pc = [
        b"Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/56.0.2924.87 Safari/537.36",
        b"Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; AcooBrowser; .NET CLR 1.1.4322; .NET CLR 2.0.50727)",
        b"Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0; Acoo Browser; SLCC1; .NET CLR 2.0.50727; "
        b"Media Center PC 5.0; .NET CLR 3.0.04506)",
        b"Mozilla/4.0 (compatible; MSIE 7.0; AOL 9.5; AOLBuild 4337.35; Windows NT 5.1; .NET CLR 1.1.4322; "
        b".NET CLR 2.0.50727)",
        b"Mozilla/5.0 (Windows; U; MSIE 9.0; Windows NT 9.0; en-US)",
        b"Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Win64; x64; Trident/5.0; .NET CLR 3.5.30729; "
        b".NET CLR 3.0.30729; .NET CLR 2.0.50727; Media Center PC 6.0)",
        b"Mozilla/5.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0; WOW64; Trident/4.0; SLCC2; "
        b".NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; .NET CLR 1.0.3705; .NET CLR 1.1.4322)",
        b"Mozilla/4.0 (compatible; MSIE 7.0b; Windows NT 5.2; .NET CLR 1.1.4322; .NET CLR 2.0.50727; InfoPath.2; "
        b".NET CLR 3.0.04506.30)",
        b"Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN) AppleWebKit/523.15 (KHTML, like Gecko, Safari/419.3)"
        b" Arora/0.3 (Change: 287 c9dfb30)",
        b"Mozilla/5.0 (X11; U; Linux; en-US) AppleWebKit/527+ (KHTML, like Gecko, Safari/419.3) Arora/0.6",
        b"Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.8.1.2pre) Gecko/20070215 K-Ninja/2.1.1",
        b"Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN; rv:1.9) Gecko/20080705 Firefox/3.0 Kapiko/3.0",
        b"Mozilla/5.0 (X11; Linux i686; U;) Gecko/20070322 Kazehakase/0.4.5",
        b"Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.0.8) Gecko Fedora/1.9.0.8-1.fc10 Kazehakase/0.5.6",
        b"Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
        b"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_3) AppleWebKit/535.20 (KHTML, like Gecko) "
        b"Chrome/19.0.1036.7 Safari/535.20",
    ]
    user_agent_mobile = []

    def __init__(self, settings, user_agent='Scrapy'):
        self.user_agent = user_agent
        self.logger = Logging()
        self._keymsg = key_business_param_from_settings(settings)
        self._keymsg['content'] = 'UserAgentMiddleware initialized'
        self.logger.info(json.dumps(self._keymsg))

    def process_request(self, request, spider):
        _ua = request.headers.get('User-Agent', '')
        if not _ua or b'scrapy' in _ua.lower():
            _pre_ua = request.headers.get('User-Agent')
            request.headers['User-Agent'] = random.choice(self.user_agent_pc)
            self.logger.debug(f"UA changed from {_pre_ua} to {request.headers['User-Agent']}")
        else:
            self.logger.debug(f"UA not changed <{request.headers['User-Agent']}")

    @classmethod
    def from_settings(cls, settings):
        ua = settings.get('USER_AGENT', '')
        return cls(settings, ua)
